package net.zoneland.zrdp.system.service.impl;

import java.util.*;
import java.util.stream.Collectors;

import com.richercloud.lift.util.function.enhancer.FunctionEnhancer;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;

import net.zoneland.zrdp.common.constant.Constants;
import net.zoneland.zrdp.common.constant.UserConstants;
import net.zoneland.zrdp.common.core.domain.TreeSelect;
import net.zoneland.zrdp.common.core.domain.entity.SysMenu;
import net.zoneland.zrdp.common.core.domain.entity.SysRole;
import net.zoneland.zrdp.common.core.domain.entity.SysUser;
import net.zoneland.zrdp.common.utils.SecurityUtils;
import net.zoneland.zrdp.common.utils.StringUtilPlus;
import net.zoneland.zrdp.system.domain.vo.MetaVo;
import net.zoneland.zrdp.system.domain.vo.RouterVo;
import net.zoneland.zrdp.system.mapper.SysMenuMapper;
import net.zoneland.zrdp.system.mapper.SysRoleMapper;
import net.zoneland.zrdp.system.mapper.SysRoleMenuMapper;
import net.zoneland.zrdp.system.service.ISysMenuService;

import javax.annotation.Resource;


/**
 * 菜单 业务层处理
 *
 * @author zonevue
 */
@Service
public class SysMenuServiceImpl implements ISysMenuService {

    @Resource
    private SysMenuMapper menuMapper;

    @Resource
    private SysRoleMapper roleMapper;

    @Resource
    private SysRoleMenuMapper roleMenuMapper;


    /**
     * 根据用户查询系统菜单列表
     *
     * @param userId 用户ID
     * @return 菜单列表
     */
    @Override
    public List<SysMenu> selectMenuList(final Long userId) {
        return selectMenuList(new SysMenu(), userId);
    }

    /**
     * 查询系统菜单列表
     *
     * @param menu 菜单信息
     * @return 菜单列表
     */
    @Override
    public List<SysMenu> selectMenuList(final SysMenu menu, final Long userId) {
        // 管理员显示所有菜单信息
        if (SysUser.isAdmin(userId)) {
            return menuMapper.selectMenuList(menu);
        }
        menu.getParams().put("userId", userId);
        return menuMapper.selectMenuListByUserId(menu);

    }

    /**
     * 根据用户ID查询权限
     *
     * @param userId 用户ID
     * @return 权限列表
     */
    @Override
    public Set<String> selectMenuPermsByUserId(final Long userId) {
        final List<String> perms = menuMapper.selectMenuPermsByUserId(userId);
        return menuListToSet(perms);
    }


    /**
     * 根据角色ID查询权限
     *
     * @param roleId 角色ID
     * @return 权限列表
     */
    @Override
    public Set<String> selectMenuPermsByRoleId(final Long roleId) {
        final List<String> perms = menuMapper.selectMenuPermsByRoleId(roleId);
        return menuListToSet(perms);
    }

    private Set<String> menuListToSet(final List<String> perms) {
        return perms.stream()
            .filter(StringUtils::isNotEmpty)
            .map(String::trim)
            .map(perm -> StringUtils.split(perm, ","))
            .flatMap(Arrays::stream)
            .collect(Collectors.toSet());
    }

    /**
     * 根据用户ID查询菜单
     *
     * @param userId 用户名称
     * @return 菜单列表
     */
    @Override
    public List<SysMenu> selectMenuTreeByUserId(final Long userId) {
        final List<SysMenu> menus = SecurityUtils.isAdmin(userId) ?
            menuMapper.selectMenuTreeAll() : menuMapper.selectMenuTreeByUserId(userId);

        return getChildPerms(menus, 0);
    }


    /**
     * 根据角色ID查询菜单树信息
     *
     * @param roleId 角色ID
     * @return 选中菜单列表
     */
    @Override
    public List<Long> selectMenuListByRoleId(final Long roleId) {
        final SysRole role = roleMapper.selectRoleById(roleId);
        return menuMapper.selectMenuListByRoleId(roleId, role.isMenuCheckStrictly());
    }

    /**
     * 构建前端路由所需要的菜单
     *
     * @param menus 菜单列表
     * @return 路由列表
     */
    @Override
    public List<RouterVo> buildMenus(final List<SysMenu> menus) {
        return menus.stream().map(menu -> {
            final RouterVo router = new RouterVo();
            router.setHidden("1".equals(menu.getVisible()));
            router.setName(getRouteName(menu));
            router.setPath(getRouterPath(menu));
            router.setComponent(getComponent(menu));
            router.setQuery(menu.getQuery());
            router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));

            final List<SysMenu> cMenus = menu.getChildren();
            if (CollectionUtils.isNotEmpty(cMenus) && UserConstants.TYPE_DIR.equals(menu.getMenuType())) {
                router.setAlwaysShow(true);
                router.setRedirect("noRedirect");
                router.setChildren(buildMenus(cMenus));
            } else if (isMenuFrame(menu)) {
                router.setMeta(null);
                final RouterVo children = new RouterVo();
                children.setPath(menu.getPath());
                children.setComponent(menu.getComponent());
                children.setName(StringUtils.capitalize(menu.getPath()));
                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), StringUtils.equals("1", menu.getIsCache()), menu.getPath()));
                children.setQuery(menu.getQuery());
                router.setChildren(Collections.singletonList(children));
            } else if (menu.getParentId() == 0 && isInnerLink(menu)) {
                router.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon()));
                router.setPath("/");
                final RouterVo children = new RouterVo();
                final String routerPath = innerLinkReplaceEach(menu.getPath());
                children.setPath(routerPath);
                children.setComponent(UserConstants.INNER_LINK);
                children.setName(StringUtils.capitalize(routerPath));
                children.setMeta(new MetaVo(menu.getMenuName(), menu.getIcon(), menu.getPath()));
                router.setChildren(Collections.singletonList(children));
            }

            return router;
        }).collect(Collectors.toList());
    }


    /**
     * 构建前端所需要树结构
     *
     * @param menus 菜单列表
     * @return 树结构列表
     */
    @Override
    public List<SysMenu> buildMenuTree(final List<SysMenu> menus) {
        final Set<Long> tempList = menus.stream().map(SysMenu::getMenuId).collect(Collectors.toSet());
        final List<SysMenu> returnList = menus.stream()
            // 找出所有一级菜单（考虑到数据隔离，传进来的menu可能是部分数据）
            .filter(FunctionEnhancer.with(SysMenu::getParentId).thenTest(tempList::contains).negate())
            .map(menu -> {
                recursionFn(menus, menu);
                return menu;
            })
            .collect(Collectors.toList());

        return returnList.isEmpty() ? menus : returnList;
    }


    /**
     * 构建前端所需要下拉树结构
     *
     * @param menus 菜单列表
     * @return 下拉树结构列表
     */
    @Override
    public List<TreeSelect> buildMenuTreeSelect(final List<SysMenu> menus) {
        final List<SysMenu> menuTrees = buildMenuTree(menus);
        return menuTrees.stream().map(TreeSelect::new).collect(Collectors.toList());
    }

    /**
     * 根据菜单ID查询信息
     *
     * @param menuId 菜单ID
     * @return 菜单信息
     */
    @Override
    public SysMenu selectMenuById(final Long menuId) {
        return menuMapper.selectMenuById(menuId);
    }

    /**
     * 是否存在菜单子节点
     *
     * @param menuId 菜单ID
     * @return 结果
     */
    @Override
    public boolean hasChildByMenuId(final Long menuId) {
        return menuMapper.hasChildByMenuId(menuId);
    }

    /**
     * 查询菜单使用数量
     *
     * @param menuId 菜单ID
     * @return 结果
     */
    @Override
    public boolean checkMenuExistRole(final Long menuId) {
        return roleMenuMapper.checkMenuExistRole(menuId);
    }

    /**
     * 新增保存菜单信息
     *
     * @param menu 菜单信息
     * @return 结果
     */
    @Override
    public int insertMenu(final SysMenu menu) {
        return menuMapper.insertMenu(menu);
    }

    /**
     * 修改保存菜单信息
     *
     * @param menu 菜单信息
     * @return 结果
     */
    @Override
    public int updateMenu(final SysMenu menu) {
        return menuMapper.updateMenu(menu);
    }

    /**
     * 删除菜单管理信息
     *
     * @param menuId 菜单ID
     * @return 结果
     */
    @Override
    public int deleteMenuById(final Long menuId) {
        return menuMapper.deleteMenuById(menuId);
    }

    /**
     * 校验菜单名称是否唯一
     *
     * @param menu 菜单信息
     * @return 结果 是否唯一
     */
    @Override
    public boolean checkMenuNameUnique(final SysMenu menu) {
        final long menuId = Objects.isNull(menu.getMenuId()) ? -1L : menu.getMenuId();
        final SysMenu info = menuMapper.selectOneByNameAndParentId(menu.getMenuName(), menu.getParentId());
        return info == null || info.getMenuId() == menuId;
    }

    /**
     * 获取路由名称
     *
     * @param menu 菜单信息
     * @return 路由名称
     */
    public String getRouteName(final SysMenu menu) {
        // 非外链并且是一级目录（类型为目录）返回 empty
        return isMenuFrame(menu) ? StringUtils.EMPTY : StringUtils.capitalize(menu.getPath());
    }

    private static final String DEFAULT_ROUTER_PATH = "/";
    /**
     * 获取路由地址
     *
     * @param menu 菜单信息
     * @return 路由地址
     */
    public String getRouterPath(final SysMenu menu) {
        final int parentId = menu.getParentId().intValue();
        final String menuType = menu.getMenuType();
        final String isFrame = menu.getIsFrame();

        // 非外链并且是一级目录（类型为目录）
        if (parentId == 0 && UserConstants.TYPE_DIR.equals(menuType) && UserConstants.NO_FRAME.equals(isFrame)) {
            return DEFAULT_ROUTER_PATH + menu.getPath();
        }
        // 非外链并且是一级目录（类型为菜单）
        else if (isMenuFrame(menu)) {
            return DEFAULT_ROUTER_PATH;
        }

        final String routerPath = menu.getPath();
        // 内链打开外网方式
        if (parentId != 0 && isInnerLink(menu)) {
            return innerLinkReplaceEach(routerPath);
        }

        return routerPath;
    }


    /**
     * 获取组件信息
     *
     * @param menu 菜单信息
     * @return 组件信息
     */
    public String getComponent(final SysMenu menu) {
        final boolean isEmptyMenuComponent = StringUtils.isEmpty(menu.getComponent());
        if (!isEmptyMenuComponent && !isMenuFrame(menu)) {
            return menu.getComponent();
        } else if (isEmptyMenuComponent && menu.getParentId() != 0 && isInnerLink(menu)) {
            return UserConstants.INNER_LINK;
        } else if (isEmptyMenuComponent && isParentView(menu)) {
            return UserConstants.PARENT_VIEW;
        }
        return UserConstants.LAYOUT;
    }

    /**
     * 是否为菜单内部跳转
     *
     * @param menu 菜单信息
     * @return 结果
     */
    public boolean isMenuFrame(final SysMenu menu) {
        return menu.getParentId() == 0 && UserConstants.TYPE_MENU.equals(menu.getMenuType())
            && menu.getIsFrame().equals(UserConstants.NO_FRAME);
    }

    /**
     * 是否为内链组件
     *
     * @param menu 菜单信息
     * @return 结果
     */
    public boolean isInnerLink(final SysMenu menu) {
        return menu.getIsFrame().equals(UserConstants.NO_FRAME) && StringUtilPlus.isHttp(menu.getPath());
    }

    /**
     * 是否为parent_view组件
     *
     * @param menu 菜单信息
     * @return 结果
     */
    public boolean isParentView(final SysMenu menu) {
        return menu.getParentId() != 0 && UserConstants.TYPE_DIR.equals(menu.getMenuType());
    }

    /**
     * 根据父节点的ID获取所有子节点
     *
     * @param list 分类表
     * @param parentId 传入的父节点ID
     * @return String
     */
    public List<SysMenu> getChildPerms(final List<SysMenu> list, final int parentId) {
        return list.stream()
            .filter(menu -> menu.getParentId() == parentId)
            .map(menu -> {
                recursionFn(list, menu);
                return menu;
            })
            .collect(Collectors.toList());
    }


    /**
     * 递归方法将给根菜单下面挂上对应的子节点以及子节点的子节点...
     *
     * @param list 所有的节点数据
     * @param root 根菜单
     */
    private void recursionFn(final List<SysMenu> list, final SysMenu root) {
        final List<SysMenu> childList = getChildList(list, root);
        root.setChildren(childList);

        childList.forEach(child -> {
            if (hasChild(list, child)) {
                recursionFn(list, child);
            }
        });
    }


    /**
     * 得到子节点列表
     */
    private List<SysMenu> getChildList(final List<SysMenu> list, final SysMenu t) {
        final Long menuId = t.getMenuId();
        return list.stream()
            .filter(n -> n.getParentId().equals(menuId))
            .collect(Collectors.toList());
    }


    /**
     * 判断是否有子节点
     */
    private boolean hasChild(final List<SysMenu> list, final SysMenu t) {
        return list.stream()
            .map(SysMenu::getParentId)
            .anyMatch(t.getMenuId()::equals);
    }

    /**
     * 内链域名特殊字符替换
     *
     * @return 替换后的内链域名
     */
    public String innerLinkReplaceEach(final String path) {
        return StringUtils.replaceEach(path, new String[] { Constants.HTTP, Constants.HTTPS, Constants.WWW, "." },
            new String[] { "", "", "", "/" });
    }
}
